#ifndef DETECT_REC_PLATE_MNN_H
#define DETECT_REC_PLATE_MNN_H

#include "lpr_infer_api.h"
#include "mnn_infer_api.h"

class DetectRecPlateMnn : public DetectRecPlate
{
private:
    mnnModel recModel;
    void get_plate_result(cv::Mat &img,std::string &plate_number, float &plate_char_conf, std::string &plate_color, std::vector<float>& char_conf);
    
public:
    DetectRecPlateMnn();
    ~DetectRecPlateMnn();
    void infer(cv::Mat &img, std::string &plate_number, float &plate_char_conf, std::string &plate_color, std::vector<float>& char_conf);
};

#endif